package com.mall.web.controller.system;

import com.mall.common.annotation.Log;
import com.mall.common.constant.UserConstants;
import com.mall.common.core.controller.BaseController;
import com.mall.common.core.domain.AjaxResult;
import com.mall.common.core.domain.Ztree;
import com.mall.common.enums.BusinessType;
import com.mall.common.utils.StringUtils;
import com.mall.framework.util.ShiroUtils;
import com.mall.system.domain.entity.SysDept;
import com.mall.system.domain.entity.SysRole;
import com.mall.system.service.SysDeptService;
import java.util.List;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

/**
 * 部门信息
 * <p>
 * Created by zhonglin on 2020/4/8.
 */
@Controller
@RequestMapping("/system/dept")
public class SysDeptController
    extends BaseController {

  private String prefix = "system/dept";

  @Autowired
  private SysDeptService deptService;

  @RequiresPermissions("system:dept:view")
  @GetMapping()
  public String dept() {

    return prefix + "/dept";
  }

  @RequiresPermissions("system:dept:list")
  @PostMapping("/list")
  @ResponseBody
  public List<SysDept> list(SysDept dept) {

    List<SysDept> deptList = deptService.selectDeptList(dept);
    return deptList;
  }

  /**
   * 新增部门
   */
  @GetMapping("/add/{parentId}")
  public String add(@PathVariable("parentId") Long parentId,
                    ModelMap mmap) {

    mmap.put("dept", deptService.selectDeptById(parentId));
    return prefix + "/add";
  }

  /**
   * 新增保存部门
   */
  @Log(title = "部门管理", businessType = BusinessType.INSERT)
  @RequiresPermissions("system:dept:add")
  @PostMapping("/add")
  @ResponseBody
  public AjaxResult addSave(@Validated SysDept dept) {

    if (UserConstants.DEPT_NAME_NOT_UNIQUE.equals(deptService.checkDeptNameUnique(dept))) {
      return error("新增部门'" + dept.getDeptName() + "'失败，部门名称已存在");
    }
    dept.setCreateBy(ShiroUtils.getLoginName());
    return toAjax(deptService.insertDept(dept));
  }

  /**
   * 修改
   */
  @GetMapping("/edit/{deptId}")
  public String edit(@PathVariable("deptId") Long deptId,
                     ModelMap mmap) {

    SysDept dept = deptService.selectDeptById(deptId);
    if (StringUtils.isNotNull(dept) && 100L == deptId) {
      dept.setParentName("无");
    }
    mmap.put("dept", dept);
    return prefix + "/edit";
  }

  /**
   * 保存
   */
  @Log(title = "部门管理", businessType = BusinessType.UPDATE)
  @RequiresPermissions("system:dept:edit")
  @PostMapping("/edit")
  @ResponseBody
  public AjaxResult editSave(@Validated SysDept dept) {

    if (UserConstants.DEPT_NAME_NOT_UNIQUE.equals(deptService.checkDeptNameUnique(dept))) {
      return error("修改部门'" + dept.getDeptName() + "'失败，部门名称已存在");
    } else if (dept.getParentId()
                   .equals(dept.getDeptId())) {
      return error("修改部门'" + dept.getDeptName() + "'失败，上级部门不能是自己");
    }
    dept.setUpdateBy(ShiroUtils.getLoginName());
    return toAjax(deptService.updateDept(dept));
  }

  /**
   * 删除
   */
  @Log(title = "部门管理", businessType = BusinessType.DELETE)
  @RequiresPermissions("system:dept:remove")
  @GetMapping("/remove/{deptId}")
  @ResponseBody
  public AjaxResult remove(@PathVariable("deptId") Long deptId) {

    if (deptService.selectDeptCount(deptId) > 0) {
      return AjaxResult.warn("存在下级部门,不允许删除");
    }
    if (deptService.checkDeptExistUser(deptId)) {
      return AjaxResult.warn("部门存在用户,不允许删除");
    }
    return toAjax(deptService.deleteDeptById(deptId));
  }

  /**
   * 校验部门名称
   */
  @PostMapping("/checkDeptNameUnique")
  @ResponseBody
  public String checkDeptNameUnique(SysDept dept) {

    return deptService.checkDeptNameUnique(dept);
  }

  /**
   * 选择部门树
   */
  @GetMapping("/selectDeptTree/{deptId}")
  public String selectDeptTree(@PathVariable("deptId") Long deptId,
                               ModelMap mmap) {

    mmap.put("dept", deptService.selectDeptById(deptId));
    return prefix + "/tree";
  }

  /**
   * 加载部门列表树
   */
  @GetMapping("/treeData")
  @ResponseBody
  public List<Ztree> treeData() {

    List<Ztree> ztrees = deptService.selectDeptTree(new SysDept());
    return ztrees;
  }

  /**
   * 加载角色部门（数据权限）列表树
   */
  @GetMapping("/roleDeptTreeData")
  @ResponseBody
  public List<Ztree> deptTreeData(SysRole role) {

    List<Ztree> ztrees = deptService.roleDeptTreeData(role);
    return ztrees;
  }

}
